square root in PYTHON